﻿#pragma once
#define _CRT_SECURE_NO_WARNINGS
#include<iostream>
#include<utility>
#include<string>
#include<vector>

using namespace std;
namespace hash_bucket
{
	template<class T>
	struct HashNode
	{
		T _data;
		HashNode<T>* _next;
		HashNode(const T& data)
			:_data(data)
			, _next(nullptr)
		{}
	};

	template<class K>
	struct MyHashFunc {
		size_t operator()(const K& key) {
			return key;
		}
	};

	template<>
	struct MyHashFunc<string> {
		size_t operator()(const string& key) {
			size_t r = 0;
			for (auto x : key) {
				r *= 13;
				r += x;
			}
			return r;
		}
	};

	// K 为 T 中key的类型
	// T 可能是键值对，也可能是K
	// KeyOfT: 从T中提取key
	// Hash将key转化为整形，因为哈市函数使用除留余数法
	template<class K, class T, class KeyOfT, class Hash = MyHashFunc<K>>
	class HashTable
	{
		typedef HashNode<T> Node;
	public:
		HashTable()
		{
			_tables.resize(10, nullptr);
		}

		// 哈希桶的销毁
		~HashTable() {
			for (int i = 0; i < _tables.size(); i++) {
				Node* cur = _tables[i];
				while (cur) {
					Node* next = cur->_next;
					delete cur;
					cur = next;
				}
				_tables[i] = nullptr;
			}

		}

		// 插入值为data的元素，如果data存在则不插入
		bool Insert(const T& data) {
			KeyOfT KeyFun;
			K key = KeyFun(data);
			if (Find(key)) {
				return false;
			}

			Hash HashFunc;
			if (_n == _tables.size()) {
				vector<Node*> NewTables(_tables.size() * 3,nullptr);

				for (int i = 0; i < _tables.size(); i++) {
					Node* cur = _tables[i];
					while (cur) {
						Node* next = cur->_next;
						size_t CurValue = HashFunc(KeyFun(cur->_data)) % NewTables.size();
						cur->_next = NewTables[CurValue];
						NewTables[CurValue]= cur;
						cur = next;
					}
					_tables[i] = nullptr;
				}
				_tables.swap(NewTables);
			}

			size_t value = HashFunc(key) % _tables.size();
			Node* NewNode = new Node(data);
			NewNode->_next = _tables[value];
			_tables[value] = NewNode;
			++_n;
			return true;
		}

		// 在哈希桶中查找值为key的元素，存在返回true否则返回false﻿
		bool Find(const K& key) {
			Hash HashFunc;
			size_t value = HashFunc(key)%_tables.size();

			Node* cur = _tables[value];
			KeyOfT KeyFunc;
			while (cur) {
				if (KeyFunc(cur->_data) == key) {
					return true;
				}
				cur = cur->_next;
			}
			return false;
		}

		// 哈希桶中删除key的元素，删除成功返回true，否则返回false
		bool Erase(const K& key) {
			if (!Find(key)) {
				return false;
			}

			Hash HashFunc;
			size_t value = HashFunc(key);
			Node* cur = _tables[value];
			Node* pre= nullptr;
			KeyOfT KeyFunc;
			while (cur) {
				if (KeyFunc(cur->_data) == key) {
					Node* del = cur;
					pre->_next = cur->_next;
					delete del;
					del = nullptr;
					--_n;
					return true;
				}
					cur = cur->_next;
					pre = cur;
			}
			return false;
		}

		const vector<Node*>& GetVector() {
			return _tables;
		}

	private:
		vector<Node*> _tables;  // 指针数组
		size_t _n = 0;			// 表中存储数据个数
	};
}
